package com.zxy.leetcode._00500_00599._00520_00529;

/**
 * https://leetcode-cn.com/problems/detect-capital/
 *
 * 检测大写字母
 *
 * 给定一个单词，你需要判断单词的大写使用是否正确。
 *
 * 我们定义，在以下情况时，单词的大写用法是正确的：
 *
 * 全部字母都是大写，比如"USA"。
 * 单词中所有字母都不是大写，比如"leetcode"。
 * 如果单词不只含有一个字母，只有首字母大写， 比如 "Google"。
 * 否则，我们定义这个单词没有正确使用大写字母。
 */
public class Test00520 {

    public static void main(String[] args) {
        Test00520 test = new Test00520();
        System.out.println(test.detectCapitalUse("USA"));
        System.out.println(test.detectCapitalUse("abc"));
        System.out.println(test.detectCapitalUse("FlaG"));
    }

    public boolean detectCapitalUse(String word) {
        if (word.length() == 1) {
            return true;
        }

        boolean first = word.charAt(0) >= 'A' && word.charAt(0) <= 'Z';
        boolean second = word.charAt(1) >= 'A' && word.charAt(1) <= 'Z';
        if (!first && second) {
            return false;
        }

        for (int i=2; i<word.length(); i++) {
            char c = word.charAt(i);
            boolean flag = (c >= 'A' && c <= 'Z');
            if (first && second && !flag) {
                return false;
            }
            if (first && !second && flag) {
                return false;
            }
            if (!first && !second && flag) {
                return false;
            }
        }

        return true;
    }
}
